# -*- coding: utf-8 -*-
# flake8: noqa

from qiniu import Auth, put_file, etag, put_data
import qiniu.config

#需要填写你的 Access Key 和 Secret Key
access_key = 'zKVmmHRE7G1E0eD3QVuaxOJ8J9dtiiD30tQXo-OK'
secret_key = 'U3rPwu5S2Hu0I57KEFIEJeXajniofDMi5qIwMKde'
def image_storage(image_data):
    #构建鉴权对象
    q = Auth(access_key, secret_key)

    #要上传的空间
    bucket_name = 'db-flask-project'

    #上传到七牛后保存的文件名
    key = None

    #生成上传 Token，可以指定过期时间等
    token = q.upload_token(bucket_name, key, 3600)

    #要上传文件的本地路径
    # localfile = '../static/news/images/person01.png'

    # ret, info = put_file(token, key, localfile)
    ret, info = put_data(token, key, image_data)
    print(info)
    print(ret)
    # assert ret['key'] == key
    # assert ret['hash'] == etag(localfile)
    return ret.get('key') if info.status_code == 200 else None

if __name__ == '__main__':
    with open('./person01.png', 'rb') as f:
        ret = image_storage(f.read())
